package com.shujia.window

import org.apache.flink.streaming.api.scala._
import org.apache.flink.streaming.api.windowing.assigners.{EventTimeSessionWindows, ProcessingTimeSessionWindows}
import org.apache.flink.streaming.api.windowing.time.Time

object Demo2SessionWindow {
  def main(args: Array[String]): Unit = {
    val env: StreamExecutionEnvironment = StreamExecutionEnvironment.getExecutionEnvironment

    env.setParallelism(1)

    //读取时间
    val linesDS: DataStream[String] = env.socketTextStream("master", 8888)


    //取出时间字段,时间字段需要转换成long类型
    val eventTimeDS: DataStream[(String, Long)] = linesDS.map(line => {
      val split: Array[String] = line.split(",")
      (split(0), split(1).toLong)
    })

    val assDS: DataStream[(String, Long)] = eventTimeDS.assignAscendingTimestamps(_._2)

    /**
      * 统计单词的数量，如果一个单词5秒没有数据产生，同情前面这个单词处理的数量
      *
      */

    /**
      * 会话窗口：同一个key多久没有数据开始计算
      *
      * ProcessingTimeSessionWindows：处理时间的会话窗口
      * EventTimeSessionWindows： 事件实践的会话窗口， 设置时间字段,  当下一次数据时间减去上一次数据的时间大于5秒才计算
      *
      */

    val kvDS: DataStream[(String, Int)] = assDS.map(kv => (kv._1, 1))

    kvDS
      .keyBy(_._1)
      .window(EventTimeSessionWindows.withGap(Time.seconds(5)))
      .sum(1)
      .print()

    env.execute()
  }

}
